"""
数据库工具类
作者：资深Flask讲师
功能：数据库初始化和管理
"""

from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

# 创建SQLAlchemy实例
db = SQLAlchemy()

# 创建Migrate实例（用于数据库迁移）
migrate = Migrate()


def init_db(app):
    """
    初始化数据库
    :param app: Flask应用实例
    """
    # 初始化SQLAlchemy
    db.init_app(app)
    
    # 初始化Migrate
    migrate.init_app(app, db)
    
    print("[SUCCESS] 数据库初始化成功")
    
    return db

